import json
import os
import time

from stardog.cloud import Client
from stardog.cloud.exceptions import RateLimitExceededException, StardogCloudException

app_api_token = os.getenv("VOICEBOX_APP_TOKEN")

try:
    results = {}
    with Client(base_url="https://portal.stardoglabs.com/api") as client:
        program_start = time.time()
        app = client.voicebox_app(
            app_api_token=app_api_token, client_id="your-client-id"
        )
        for i in range(2):
            start = time.time()
            response = app.ask(question="How many sensors are there?")
            end = time.time()
            results[i] = {**response.model_dump(), "response_time": f"{end-start:.2f}"}
        program_end = time.time()
        print(json.dumps(results, indent=2))
        print(f"Total time taken: {program_end-program_start:.2f}")

except RateLimitExceededException as e:
    print("Rate limit error.")
except StardogCloudException as e:
    print(f"Error occurred: {e}")

import asyncio
import json
import os
import time

from stardog.cloud import AsyncClient
from stardog.cloud.exceptions import RateLimitExceededException, StardogCloudException
from stardog.cloud.voicebox import VoiceboxApp

app_api_token = os.getenv("VOICEBOX_APP_TOKEN")

results = {}


async def ask_and_persist(app: VoiceboxApp, index: int, question: str):
    start = time.time()
    answer = await app.async_ask(question)
    print(answer)
    end = time.time()
    results[index] = {**answer.model_dump(), "response_time": f"{end-start:.2f}"}


async def benchmark():
    try:
        async with AsyncClient(base_url="https://portal.stardoglabs.com/api") as client:
            app = client.voicebox_app(
                app_api_token=app_api_token, client_id="your-client-id"
            )
            tasks = [
                ask_and_persist(app, i, "How many sensors are there?") for i in range(2)
            ]
            program_start = time.time()
            await asyncio.gather(*tasks)
            program_end = time.time()
            print(json.dumps(results, indent=2))
            print(f"Total time taken: {program_end-program_start:.2f}")

    except RateLimitExceededException as e:
        print("Rate limit error.")
    except StardogCloudException as e:
        print(f"Error occurred: {e}")


asyncio.run(benchmark())